# async_sessionmaker import from sqlalchemy.ext.asyncio
AsyncSessionFactory = async_sessionmaker(engine, class_=AsyncSession, expire_on_commit=False)

async def get_async_session() -> AsyncGenerator[AsyncSession, None]:
    async with AsyncSessionFactory() as session:
        yield session

@asynccontextmanager
async def lifespan(_fastapi: FastAPI):
    if broker.is_worker_process:
        # TaskIQ configurartion so we can share FastAPI dependencies in tasks
        await broker.startup()
    
    yield

    if broker.is_worker_process:
        # On shutdown, we need to shutdown the broker
        await broker.shutdown()

# Custom factory to be able to set the sender globally
class EmailSenderFactory(EmailSender):
    @property
    def sender(self):
        return self.sender
    
    @sender.setter
    def sender(self, sender: str):
        self.sender = sender